"""
mainpage.py


"""
from common.basepage import BasePage
from common.loginpage import LoginPage


class MainPage(BasePage):
    def __init__(self,driver=None):
        super().__init__(driver)
        # 登录页面用到的元素的定位方式
        # 后台管理按钮的定位方式
        self.back_manage_loc ="xpath=/html/body/div[1]/div[1]/div/ul[1]/li[8]/button/i"
        # 后天管理页面的框架名称
        self.back_manage_frame_name = "iframe-superadmin"
        # 签退 超链接的定位方式
        self.quit_loc="link text=签退"
        self.assert_text_loc = "link text=签退"
    # 点击后台管理按钮方法
    def type_back_mange(self):
        self.type_link(self.back_manage_loc)
        self.toframe(self.back_manage_frame_name)
    # 点击签退超连接
    def type_quit_link(self):
        self.driver.switch_to.default_content()
        self.type_link(self.quit_loc)
    # 获取当前页面的断言文本
    def get_mainpage_assert_text(self):
        self.driver.switch_to.default_content()
        return self.get_text(self.assert_text_loc)

    def test_mainpage(self):
        lp = LoginPage(self.driver)
        lp.test_login()
        self.type_back_mange()
        self.wait(2)
        print(self.get_mainpage_assert_text())
        self.wait(2)
        self.type_quit_link()
        self.wait(2)

if __name__ == '__main__':

    mp = MainPage()

    mp.test_mainpage()



